import onnx
from onnx import helper, TensorProto

X = helper.make_tensor_value_info("x", TensorProto.FLOAT, [2, 2])
Y = helper.make_tensor_value_info("y", TensorProto.FLOAT, [2, 2])
Z = helper.make_tensor_value_info("z", TensorProto.FLOAT, [2, 2])

add = helper.make_node("Add", inputs=["x", "y"], outputs=["z"], name="add0")

graph = helper.make_graph([add], "add_graph", [X, Y], [Z])
model = helper.make_model(graph, opset_imports=[helper.make_operatorsetid("", 13)])

onnx.checker.check_model(model)
onnx.save(model, "add_v13.onnx")
print("=> wrote add_v13.onnx")
